Journals
  Publication Years
  Keywords
Search within results Open Search
Please wait a minute...
For Selected: Toggle Thumbnails
Improved algorithm of generative adversarial network based on arbitration mechanism
Guihui CHEN, Huikang LIU, Zhongbing LI, Jiao PENG, Shaotian WANG, Jinyu LIN
Journal of Computer Applications    2021, 41 (11): 3185-3191.   DOI: 10.11772/j.issn.1001-9081.2020122040
Abstract332)   HTML13)    PDF (2958KB)(97)       Save

Concerning the lack of flexibility in adversarial training of Deep Convolutional Generative Adversarial Network (DCGAN) and the problems of inflexible optimization and unclear convergence state of Binary Cross-Entropy loss (BCE loss) function used in DCGAN, an improved algorithm of Generative Adversarial Network (GAN) based on arbitration mechanism was proposed. In this algorithm, the proposed arbitration mechanism was added on the basis of DCGAN. Firstly, the network structure of the proposed improved algorithm was composed of generator, discriminator, and arbiter. Secondly, the adversarial training was conducted by the generator and discriminator according to the training plan, and the abilities to generate images and verify the authenticity of images were strengthened according to the characteristics learned from the dataset respectively. Thirdly, the arbiter was generated by the generator and the discriminator after the last round of adversarial training and metric score calculation module, and the adversarial training results of the generator and the discriminator were measured by this arbiter and fed back into the training plan. Finally, a wining limit was added to the network structure to improve the stability of model training, and the Circle loss function was used to replace the BCE loss function, which made the model optimization process more flexible and the convergence state more clear. Experimental results show that the proposed algorithm has a good generation effect on the architectural and face datasets. On the Large-scale Scene UNderstanding (LSUN) dataset, the proposed algorithm has the Fréchet Inception Distance (FID) index decreased by 1.04% compared with the DCGAN original algorithm; on the CelebA dataset, the proposed algorithm has the Inception Score (IS) index increased by 4.53% compared with the DCGAN original algorithm. The images generated by the proposed algorithm have better diversity and higher quality.

Table and Figures | Reference | Related Articles | Metrics